package com.campsys.controller;

import com.campsys.core.Result;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.nio.file.Paths;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;

@Api(tags = "通用功能")
@RestController
@RequestMapping("/common")
@Slf4j
public class CommonController {

    private static final String FILE_UPLOAD_DIR = Paths.get("src", "main", "resources", "file").toString();

    @ApiOperation("上传文件")
    @PostMapping("/upload")
    public Result uploadFile(@RequestParam("file") MultipartFile file) {
        if (file.isEmpty()) {
            return Result.error("上传失败，请选择文件");
        }

        try {
            // 获取文件名
            String fileName = file.getOriginalFilename();
            // 获取文件扩展名
            String fileExtension = "";
            if (fileName.contains(".")) {
                int lastIndex = fileName.lastIndexOf('.');
                fileExtension = fileName.substring(lastIndex);
                fileName = fileName.substring(0, lastIndex);
            }
            // 生成新文件名（原文件名 + 当前时间戳）
            String newFileName = fileName + "_" + LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMddHHmmssSSS")) + fileExtension;
            // 创建文件存储目录（如果不存在）
            File uploadDir = new File(FILE_UPLOAD_DIR);
            if (!uploadDir.exists()) {
                uploadDir.mkdirs();
            }
            // 获取项目根目录
            String rootPath = System.getProperty("user.dir");
            // 设置文件存储路径
            String filePath = rootPath + File.separator + FILE_UPLOAD_DIR + File.separator + newFileName;
            // 将文件保存到指定路径
            file.transferTo(new File(filePath));
            String relativePath = Paths.get(FILE_UPLOAD_DIR, newFileName).toString();
            return Result.success("上传成功",relativePath);
        } catch (IOException e) {
            e.printStackTrace();
            return Result.error("上传失败");
        }
    }

    @ApiOperation("下载文件")
    @PostMapping("/download")
    public void download(String path, HttpServletResponse response) {
        try {
            // 获取项目根目录
            String rootPath = System.getProperty("user.dir");
            // 设置文件存储路径
            String filePath = rootPath + File.separator + path;
            // path是指想要下载的文件的路径
            File file = new File(filePath);
            log.info(file.getPath());
            // 获取文件名
            String filename = file.getName();
            // 获取文件后缀名
            String ext = filename.substring(filename.lastIndexOf(".") + 1).toLowerCase();
            log.info("文件后缀名：" + ext);

            // 将文件写入输入流
            FileInputStream fileInputStream = new FileInputStream(file);
            InputStream fis = new BufferedInputStream(fileInputStream);
            byte[] buffer = new byte[fis.available()];
            fis.read(buffer);
            fis.close();

            // 清空response
            response.reset();
            // 设置response的Header
            response.setCharacterEncoding("UTF-8");
            //Content-Disposition的作用：告知浏览器以何种方式显示响应返回的文件，用浏览器打开还是以附件的形式下载到本地保存
            //attachment表示以附件方式下载   inline表示在线打开   "Content-Disposition: inline; filename=文件名.mp3"
            // filename表示文件的默认名称，因为网络传输只支持URL编码的相关支付，因此需要将文件名URL编码后进行传输,前端收到后需要反编码才能获取到真正的名称
            response.addHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filename, "UTF-8"));
            // 告知浏览器文件的大小
            response.addHeader("Content-Length", "" + file.length());
            OutputStream outputStream = new BufferedOutputStream(response.getOutputStream());
            response.setContentType("application/octet-stream");
            outputStream.write(buffer);
            outputStream.flush();
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

}
